from langchain_openai import ChatOpenAI
from browser_use import Agent
import asyncio, os
from dotenv import load_dotenv
# 加载环境变量
load_dotenv()


async def main():
    # 初始化模型
    llm = ChatOpenAI(
        base_url="https://api.siliconflow.cn/v1",
        model="deepseek-ai/DeepSeek-V3",
        api_key=os.getenv('API_KEY')
    )
    # print(llm)

    # 创建Agent并定义UI测试任务
    agent = Agent(
        task="""
        1. 访问百度首页（https://www.baidu.com/）
        2. 点击导航栏的“新闻”菜单
        3. 列出“热点要闻”板块中前十条新闻的标题和发布时间,
        用json格式返回结果，示范格式如下：
        [ { title: '时尚板块最新新闻标题1', publish_time: '3月3日 13:20'}]
        """,
        llm=llm,
        use_vision=False  # 禁用视觉模式，依赖DOM解析
    )
    # 执行任务后输出结果
    result = await agent.run()
    print("\n======== 测试结果 ========")
    print(result.final_result())

asyncio.run(main())